package cn.iocoder.yudao.module.ym.service.tickettrajectory;

import cn.iocoder.yudao.module.infra.enums.TicketStatusEnum;
import cn.iocoder.yudao.module.ym.controller.admin.site.vo.SitePageReqVO;
import cn.iocoder.yudao.module.ym.controller.admin.ticket.vo.TicketPageReqVO;
import cn.iocoder.yudao.module.ym.dal.dataobject.area.AreaDO;
import cn.iocoder.yudao.module.ym.dal.dataobject.site.SiteDO;
import cn.iocoder.yudao.module.ym.dal.dataobject.ticket.TicketDO;
import cn.iocoder.yudao.module.ym.dal.mysql.site.SiteMapper;
import cn.iocoder.yudao.module.ym.dal.mysql.ticket.TicketMapper;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import org.apache.commons.math3.analysis.function.Identity;
import org.springframework.stereotype.Service;
import jakarta.annotation.Resource;
import org.springframework.validation.annotation.Validated;
import org.springframework.transaction.annotation.Transactional;

import java.util.*;
import java.util.function.Function;
import java.util.stream.Collectors;

import cn.iocoder.yudao.module.ym.controller.admin.tickettrajectory.vo.*;
import cn.iocoder.yudao.module.ym.dal.dataobject.tickettrajectory.TicketTrajectoryDO;
import cn.iocoder.yudao.framework.common.pojo.PageResult;
import cn.iocoder.yudao.framework.common.pojo.PageParam;
import cn.iocoder.yudao.framework.common.util.object.BeanUtils;

import cn.iocoder.yudao.module.ym.dal.mysql.tickettrajectory.TicketTrajectoryMapper;

import static cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil.exception;
import static cn.iocoder.yudao.module.infra.enums.ErrorCodeConstants.TICKET_ILLEGAL;
import static cn.iocoder.yudao.module.infra.enums.ErrorCodeConstants.TICKET_TRAJECTORY_NOT_EXISTS;

/**
 * 工单轨迹 Service 实现类
 *
 * @author 韩艳阳
 */
@Service
@Validated
public class TicketTrajectoryServiceImpl implements TicketTrajectoryService {

    @Resource
    private TicketTrajectoryMapper ticketTrajectoryMapper;

    @Resource
    private TicketMapper ticketMapper;

    @Resource
    private SiteMapper siteMapper;


    @Override
    public Long createTicketTrajectory(TicketTrajectorySaveReqVO createReqVO) {
        //判断该工单是否合法
        Long ticketId = createReqVO.getTicketId();

        //0跳过
        if(ticketId != 0){
            TicketDO ticketDO = ticketMapper.selectById(ticketId);
            if(ticketDO == null || (ticketDO.getTicketStatus() != TicketStatusEnum.PROCESSING.getCode() && ticketDO.getTicketStatus() != TicketStatusEnum.TIMEOUT.getCode())){
                throw exception(TICKET_ILLEGAL);
            }
        }

        // 插入
        TicketTrajectoryDO ticketTrajectory = BeanUtils.toBean(createReqVO, TicketTrajectoryDO.class);
        ticketTrajectoryMapper.insert(ticketTrajectory);
        // 返回
        return ticketTrajectory.getId();
    }

    @Override
    public void updateTicketTrajectory(TicketTrajectorySaveReqVO updateReqVO) {
        // 校验存在
        validateTicketTrajectoryExists(updateReqVO.getId());
        // 更新
        TicketTrajectoryDO updateObj = BeanUtils.toBean(updateReqVO, TicketTrajectoryDO.class);
        ticketTrajectoryMapper.updateById(updateObj);
    }

    @Override
    public void deleteTicketTrajectory(Long id) {
        // 校验存在
        validateTicketTrajectoryExists(id);
        // 删除
        ticketTrajectoryMapper.deleteById(id);
    }

    /**
     * 获取某人一天的轨迹
     * @param userId
     * @return
     */
    @Override
    public Map<String, Object> getDayByUserId(Long userId,String date) {
        Map<String, Object> map = new HashMap<>();

        //获取今天的轨迹，先不去重
        List<TicketTrajectoryDO> ticketTrajectoryDOS = ticketTrajectoryMapper.selectList(new LambdaQueryWrapper<TicketTrajectoryDO>()
                .eq(TicketTrajectoryDO::getOperatorId, userId)
                .apply("DATE(create_time) = DATE('"+date+"')"));

        if(ticketTrajectoryDOS != null && !ticketTrajectoryDOS.isEmpty()){

            List<String> collect = ticketTrajectoryDOS.stream().map(TicketTrajectoryDO::getLocation).collect(Collectors.toList());
            if(!collect.isEmpty()){
                map.put("location",collect);
            }
            Set<Long> ticketIds = ticketTrajectoryDOS.stream().map(TicketTrajectoryDO::getTicketId).collect(Collectors.toSet());

            List<TicketDO> ticketDOS = ticketMapper.selectList(new LambdaQueryWrapper<TicketDO>().in(TicketDO::getId, ticketIds));

            if(ticketDOS != null && !ticketDOS.isEmpty()){
                Set<Long> siteIds = ticketDOS.stream().map(TicketDO::getSiteId).collect(Collectors.toSet());

                SitePageReqVO sitePageReqVO = new SitePageReqVO();
                sitePageReqVO.setIds(siteIds);
                List<SiteDO> siteDOS = siteMapper.selectList(sitePageReqVO);
                if(siteDOS != null && !siteDOS.isEmpty()){
                    map.put("siteList",siteDOS);
                }
            }
        }

        return map;
    }

    private void validateTicketTrajectoryExists(Long id) {
        if (ticketTrajectoryMapper.selectById(id) == null) {
            throw exception(TICKET_TRAJECTORY_NOT_EXISTS);
        }
    }

    @Override
    public TicketTrajectoryDO getTicketTrajectory(Long id) {
        return ticketTrajectoryMapper.selectById(id);
    }

    @Override
    public List<TicketTrajectoryDO> getTicketTrajectoryList(TicketTrajectoryPageReqVO pageReqVO) {
        return ticketTrajectoryMapper.selectList(pageReqVO);
    }

    @Override
    public PageResult<TicketTrajectoryDO> getTicketTrajectoryPage(TicketTrajectoryPageReqVO pageReqVO) {
        return ticketTrajectoryMapper.selectPage(pageReqVO);
    }

}